package util;

import us.codecraft.webmagic.Page;
import us.codecraft.webmagic.Request;


public class addTargetRequest {
    public static void add(Page page,String url,String search_word,long priority) {
        /**
         * @Description 访问公司主页
         * @param url 公司页面链接
         * @param search_word 搜索词，用于拼接Referer信息
         */
        Request r = new Request(url);
        r.addHeader("Referer","www.qichacha.com/search?key="+UrlUtil.getURLEncoderString(search_word));
        r.setPriority(priority);
        r.setExtras(page.getRequest().getExtras());
        page.addTargetRequest(r);
    }
    public static void add(Page page,String unique,String company_name,String tab,long priority){
        /**
         * @Description 拼接请求,访问xhr接口
         * @param unique 企查查id
         * @param company_name 公司名称
         * @param tab 标签页
         */

        String GETINFOS_URL= "https://www.qichacha.com/company_getinfos?unique=%s&companyname=%s&tab=%s";
        String url = String.format(GETINFOS_URL,
                unique,
                company_name,
                tab
        );
        Request r = new Request(url);
        r.addHeader("X-Requested-With","XMLHttpRequest");
        r.addHeader("Accept","text/html, */*; q=0.01");
        //替换Refere参数为公司主页
        r.addHeader("Referer","https://www.qichacha.com/firm_"+unique+".html");
        r.setPriority(priority);
        r.setExtras(page.getRequest().getExtras());
        //请求request增加参数页数和公司id
        r.putExtra("unique",unique);
        r.putExtra("company_name",company_name);
        page.addTargetRequest(r);
    }
    public static void add(Page page,String unique,String company_name,String tab,String box,int page_id,long priority) {
        /**
         * @Description 拼接请求,访问分页的数据
         * @param unique 企查查id
         * @param company_name 公司名称
         * @param tab 标签页
         * @param box 分页对象
         * @param page_id 分页的页数
         */
        String GETINFOS_URL = "https://www.qichacha.com/company_getinfos?unique=%s&companyname=%s&p=%d&tab=%s&box=%s";

        String url = String.format(GETINFOS_URL,
                unique,
                company_name,
                page_id,
                tab,
                box);
        Request r = new Request(url);
        r.addHeader("X-Requested-With","XMLHttpRequest");
        r.addHeader("Accept","text/html, */*; q=0.01");
        r.setPriority(priority);
        //替换Refere参数为公司主页
        r.addHeader("Referer","https://www.qichacha.com/firm_"+unique+".html");
        //请求request增加参数页数和公司id

        r.setExtras(page.getRequest().getExtras());
        r.putExtra("unique",unique);
        r.putExtra("page_id",page_id);
        r.putExtra("company_name",company_name);
        page.addTargetRequest(r);
    }
}
